/**
 * 617. 合并二叉树
 *
 * 【简单】
 *
 * 给你两棵二叉树： root1 和 root2 。
 * 想象一下，当你将其中一棵覆盖到另一棵之上时，两棵树上的一些节点将会重叠（而另一些不会）。你需要将这两棵树合并成一棵新二叉树。合并的规则是：如果两个节点重叠，那么将这两个节点的值相加作为合并后节点的新值；否则，不为 null 的节点将直接作为新二叉树的节点。
 * 返回合并后的二叉树。
 * 注意: 合并过程必须从两个树的根节点开始。
 *
 * 示例 1：
 * 【https://assets.leetcode.com/uploads/2021/02/05/merge.jpg】
 * 输入：root1 = [1,3,2,5], root2 = [2,1,3,null,4,null,7]
 * 输出：[3,4,5,5,4,null,7]
 *
 * 示例 2：
 * 输入：root1 = [1], root2 = [1,2]
 * 输出：[2,2]
 */

import { TreeNode } from '../utils/tree'

let mergeTrees = function mergeTrees(
	root1: TreeNode | null,
	root2: TreeNode | null
): TreeNode | null {
	if (root1 && root2) {
		root1.val += root2.val
		root1.left = mergeTrees(root1.left, root2.left)
		root1.right = mergeTrees(root1.right, root2.right)

		return root1
	}

	if (root1) {
		return root1
	}

	if (root2) {
		return root2
	}

	return null
}

// 迭代
mergeTrees = (
	root1: TreeNode | null,
	root2: TreeNode | null
): TreeNode | null => {
	if (root1 == null && root2 == null) {
		return null
	}
	if (root1 == null) {
		return root2
	}

	if (root2 == null) {
		return root1
	}

	const queue = [root1, root2]
	while (queue.length) {
		const node1 = queue.shift()!
		const node2 = queue.shift()!
		node1.val += node2.val

		if (node1.left && node2.left) {
			queue.push(node1.left)
			queue.push(node2.left)
		}

		if (node1.right && node2.right) {
			queue.push(node1.right)
			queue.push(node2.right)
		}

		if (node1.left == null && node2.left) {
			node1.left = node2.left
		}

		if (node1.right == null && node2.right) {
			node1.right = node2.right
		}
	}

	return root1
}
